import unittest
from public.public import get_db_connection,close_db_connection
from public.Tools.base_page import BasePage


class FrontDesk(unittest.TestCase):
    def setUp(self):
        print("每一个测试用例执行前,打开浏览器,打开网址")
        self.dp=BasePage()
        self.dp.open_browser()
        self.dp.get('http://172.27.40.240/ecshop')
    def tearDown(self):
        self.dp.driver.quit()
    @classmethod
    def setUpClass(cls):
        print("每一个类用例执行前,连接数据库")
        # 获取数据库连接并赋值给类属性，供整个类使用
        cls.db_conn = get_db_connection()
        if cls.db_conn:  # 仅当连接成功时创建游标
            cls.cur = cls.db_conn.cursor()
        else:
            print("数据库连接失败，无法创建游标")

    @classmethod
    def tearDownClass(cls):
        print("每一个类执行后,销毁数据库")
        if cls.cur:
            cls.cur.close()
            print("游标已关闭")
            # 关闭数据库连接（传入类属性的连接对象）
        close_db_connection(cls.db_conn)

